package main;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.Sqlite3Dialect;
import com.jfinal.plugin.activerecord.generator.BaseModelGenerator;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.activerecord.generator.ModelGenerator;
import com.jfinal.plugin.druid.DruidPlugin;
import ext.SqliteMetaBuilder;

import javax.sql.DataSource;

/**
 * 在数据库表有任何变动时，运行一下 main 方法，极速响应变化进行代码重构
 * 生成完使用之后可直接删除生成的代码包，下次生成会自动创建包目录
 *
 * 这里边的代码不需要修改，需要配置的地方有settings.properties文件及my_base_model_template.jf模板文件中的baseModel名称及包路径
 *
 */
public class _JFinalDemoGenerator {

	private static boolean useSqlite = false;

	public static DataSource getDataSource() {
		DruidPlugin druidPlugin;
		String jdbcUrl = PropKit.get("db.jdbc.url");
		if(jdbcUrl.indexOf("#PATH#") > -1){
			useSqlite = true;
			jdbcUrl = jdbcUrl.replace("#PATH#",PropKit.get("db.sqlite.path"));
			druidPlugin = new DruidPlugin(jdbcUrl, "", "");
			druidPlugin.setDriverClass("org.sqlite.JDBC");
		}else{
			druidPlugin = new DruidPlugin(PropKit.get("db.jdbc.url"), PropKit.get("db.jdbc.username"), PropKit.get("db.jdbc.password").trim());
		}
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}
	
	public static void main(String[] args) {
		PropKit.use("settings.properties");
		// base model 所使用的包名
		String baseModelPackageName = PropKit.get("base.model.packge.name");
		// base model 文件保存路径
		String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/"+baseModelPackageName.replace(".","/");
		
		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = baseModelPackageName.substring(0,baseModelPackageName.lastIndexOf("."));
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";
		
		// 创建生成器
		//Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
		//自定义baseModel
        BaseModelGenerator baseModelGenerator = new BaseModelGenerator(baseModelPackageName,baseModelOutputDir);
		baseModelGenerator.setTemplate("/my_base_model_template.jf");
		Generator generator = new Generator(getDataSource(),baseModelGenerator,new ModelGenerator(modelPackageName, baseModelPackageName+".base", modelOutputDir));
		// 设置是否生成链式 setter 方法
		generator.setGenerateChainSetter(true);
		if(useSqlite){
			generator.setDialect(new Sqlite3Dialect());
			generator.setMetaBuilder(new SqliteMetaBuilder(getDataSource())); //关键点
		}
		// 添加不需要生成的表名
		String excludeTables = PropKit.get("exclude.tables");
		generator.addExcludedTable(excludeTables.split(","));
		// 设置是否在 Model 中生成 dao 对象
		generator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		generator.setGenerateDataDictionary(true);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		//generator.setRemovedTableNamePrefixes("t_");
		// 生成
		generator.generate();
		System.out.println("数据库表生成完毕");
	}
}




